import torch
from sklearn.metrics import accuracy_score, \
    precision_score, \
    recall_score, \
    f1_score
